The name of the course is Introduction to Open Data Science and we are focusing to language R, RStudio, GitHub and Markdown. You can find my Github repository here.
After the data wrangling exercise the new data set is found from the data folder. The set is based on data that was collected from course Introduction to Social Statistics, fall 2014 - in Finnish. The survey was conducted 3.12.2014 - 10.1.2015 by Kimmo Vehkalahti.
student2014 <- read.table("data/learning2014.txt", header = TRUE)
dim(student2014)
## [1] 166 7
The student data includes 7 variables and 166 rows.
str(student2014)
## 'data.frame': 166 obs. of 7 variables:
## $ gender : Factor w/ 2 levels "F","M": 1 2 1 2 2 1 2 1 2 1 ...
## $ age : int 53 55 49 53 49 38 50 37 37 42 ...
## $ attitude: int 37 31 25 35 37 38 35 29 38 21 ...
## $ deep : num 3.58 2.92 3.5 3.5 3.67 ...
## $ stra : num 3.38 2.75 3.62 3.12 3.62 ...
## $ surf : num 2.58 3.17 2.25 2.25 2.83 ...
## $ points : int 25 12 24 10 22 21 21 31 24 26 ...
Variables deep, strat and surf are combination variables from the original survey data.
summary(student2014)
## gender age attitude deep stra
## F:110 Min. :17.00 Min. :14.00 Min. :1.583 Min. :1.250
## M: 56 1st Qu.:21.00 1st Qu.:26.00 1st Qu.:3.333 1st Qu.:2.625
## Median :22.00 Median :32.00 Median :3.667 Median :3.188
## Mean :25.51 Mean :31.43 Mean :3.680 Mean :3.121
## 3rd Qu.:27.00 3rd Qu.:37.00 3rd Qu.:4.083 3rd Qu.:3.625
## Max. :55.00 Max. :50.00 Max. :4.917 Max. :5.000
## surf points
## Min. :1.583 Min. : 7.00
## 1st Qu.:2.417 1st Qu.:19.00
## Median :2.833 Median :23.00
## Mean :2.787 Mean :22.72
## 3rd Qu.:3.167 3rd Qu.:27.75
## Max. :4.333 Max. :33.00
From 166 survey respondents 56 was men and 110 was females. The mean of age was 25,5 years. Oldes respondet was 55 and youngest 17 years old.
Variables differ between genders. Distributions are different in age, attitude and surf (surfface). Three highest correlation between variables are:
##
## Call:
## lm(formula = points ~ attitude + stra + surf, data = student2014)
##
## Residuals:
## Min 1Q Median 3Q Max
## -17.1550 -3.4346 0.5156 3.6401 10.8952
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 11.01711 3.68375 2.991 0.00322 **
## attitude 0.33952 0.05741 5.913 1.93e-08 ***
## stra 0.85313 0.54159 1.575 0.11716
## surf -0.58607 0.80138 -0.731 0.46563
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 5.296 on 162 degrees of freedom
## Multiple R-squared: 0.2074, Adjusted R-squared: 0.1927
## F-statistic: 14.13 on 3 and 162 DF, p-value: 3.156e-08
In this linear regression model points are the target variable and attitude, strategy (stra) and surfface (surf) are explanatory variables.
Residuals of the model are between ~ -17.2 and ~10.9 when median is 0.52. I assume that errors are normally distributed but distribution needs confirmation.
Attitude is the only variable that has a very good significance in this model. p-value of stra and surf is too high to be even slightly significance.
Variables estimated coefficient is ~0.34 and it’s standard error is clearly smaller (~0.057). Other explanatory variables have not significance in this model.
Residual standard error is high in relation to a first and third quantiles of residuals.
This linear regression model of three explanatory variables explains ~19.3% of the points.
##
## Call:
## lm(formula = points ~ attitude, data = student2014)
##
## Residuals:
## Min 1Q Median 3Q Max
## -16.9763 -3.2119 0.4339 4.1534 10.6645
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 11.63715 1.83035 6.358 1.95e-09 ***
## attitude 0.35255 0.05674 6.214 4.12e-09 ***
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 5.32 on 164 degrees of freedom
## Multiple R-squared: 0.1906, Adjusted R-squared: 0.1856
## F-statistic: 38.61 on 1 and 164 DF, p-value: 4.119e-09
Estimated coefficient of explanatory variable attitude is ~0.35. This means that when attitude rises one point target variable (points) grows 0.35 times.
Attitude p-value (0.00000000412) shows that variable is very significant in this linear regression model
This model explains 18.6% of the points (target variable)
Residuals vs Fitted This plot shows that there is no pattern between residuals. There is a constant variance among errors. One can confirm that the assumption of constant variance of errors is valid.
Normal Q_Q Normal QQ-plot shows that the errors of the model are normally distributed.
Residual vs Leverage Last plot of the graphical model validation shows that the impact of the singel observation is moderate. Model includes some outliers but the leverage of singel observation don’t compromise the validation of the model.
Using Data Mining To Predict Secondary School Student Alcohol Consumption. Fabio Pagnotta, Hossain Mohammad Amran Department of Computer Science,University of Camerino
https://archive.ics.uci.edu/ml/datasets/STUDENT+ALCOHOL+CONSUMPTION
Data is rolled into one from Math course and Portuguese language course datasets. After the data wrangling exercise the new data set is found from the data folder.
alc <- read.table("data/student_alc.txt", header = TRUE)
dim(alc)
## [1] 382 35
The student data includes 35 variables and 382 rows.
In this analysis I’m going to study the relationship of high/low alcohol consumption between sex and the following variables:
| Variable | Type | Description |
|---|---|---|
| age | numeric | student’s age |
| studytime | numeric, scale 1-4 | weekly study time |
| freentime | numeric, scale 1-5 | free time after school |
| absence | numeric | number of school absences |
First these relationships are observed from tables and graphics. Hypothesis are as follows:
Age
The age of high consumption of alcohol may differ between sex. The development of charcter differs between young people and this may affect on habbits of alcohol consumption.
H0: Age don’t affect on alcohol consumption
H1: There is difference in between level of alcohol consumption and age
#a jitterplot of high_use, sex and age
g1 <- ggplot(alc, aes(x = high_use, y = age, col = sex))
g1 + geom_jitter() + ggtitle("Age by alcohol consumption and sex")
It seems that there is randomnes of sex and age in both alcohol consumption groups.
#a boxplot of high_use, sex and age
g1 <- ggplot(alc, aes(x = high_use, y = age, col = sex))
g1 + geom_boxplot() + ggtitle("Age by alcohol consumption and sex") + xlab("High consumption group") + ylab("Age of student")
Means however show that young male students have lower mean of age in low consumption group and females in high consumption group.
Study time
High alcohol consumption may be related to time spend in studies because one can’t do both at the same time at least not successfully.
H0: Alcohol consumption do not affect on weekly study time
H1: Alcohol consumption affects on weekly study time (numeric: 1 - <2 hours, 2 - 2 to 5 hours, 3 - 5 to 10 hours, or 4 - >10 hours)
#barplot about study time
ggplot(alc, aes(studytime, fill = high_use)) + geom_bar(position = "fill") +
ggtitle("Barplot about study time grouped by high_use")+ xlab("Study time") + ylab("Propotions of students") + scale_y_continuous(name = waiver(), breaks = waiver(), minor_breaks = waiver(), labels = waiver(), limits = NULL, expand = waiver(), na.value = NA_real_, trans = "identity")
It seems that there is less high users in those students groups who spend more time in studying (3-4) than in those who spend less time in studying (1-2).
Free time
High alcohol consumption may be related to free time so that studets who have more free time are consuming more alcohol that studets who haven’t as much free time.
H0: Amount of free time do not affect on alcohol consumption among students
H1: Amount of free time affects on alcohol consumption among students
#barplot about free time
ggplot(alc, aes(freetime, fill = high_use)) + geom_bar(position = "fill") +
ggtitle("Barplot about free time grouped by high_use")+ xlab("Free time") + ylab("Propotions of students") + scale_y_continuous(name = waiver(), breaks = waiver(), minor_breaks = waiver(), labels = waiver(), limits = NULL, expand = waiver(), na.value = NA_real_, trans = "identity")
It seems that there are more students that are consuming alcohol high amounts in those students groups that have more free time than in those who have not as much free time.
Absences
High consumption of alcohol may cause absences.
H0: High consumption of alcohol do not affect on absences
H1: High consumption of alcohol does affect on absences
#a boxplot of high_use and absences
g1 <- ggplot(alc, aes(x = high_use, y = absences, fill = high_use))
g1 + geom_boxplot() + ggtitle("Absences by alcohol consumption") + xlab("High consumption group") + ylab("Absences")
It seems that the differences between alcohol consumption groups in absences are small. Mean of absences is higher in high consumption group but it may not be significant.
#the model with glm()
m <- glm(high_use ~ age + studytime + freetime + absences, data = alc, family = "binomial")
summary(m)
##
## Call:
## glm(formula = high_use ~ age + studytime + freetime + absences,
## family = "binomial", data = alc)
##
## Deviance Residuals:
## Min 1Q Median 3Q Max
## -1.9649 -0.8267 -0.6238 1.0990 2.2861
##
## Coefficients:
## Estimate Std. Error z value Pr(>|z|)
## (Intercept) -4.39240 1.79306 -2.450 0.014299 *
## age 0.18543 0.10313 1.798 0.072183 .
## studytime -0.51842 0.15867 -3.267 0.001086 **
## freetime 0.33015 0.12430 2.656 0.007907 **
## absences 0.07856 0.02269 3.463 0.000535 ***
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## (Dispersion parameter for binomial family taken to be 1)
##
## Null deviance: 465.68 on 381 degrees of freedom
## Residual deviance: 423.18 on 377 degrees of freedom
## AIC: 433.18
##
## Number of Fisher Scoring iterations: 4
From the fitted model one can see that all explanatory variables except age are statistically significant with p-value < 0,01. Variable absence is also statistically sisgnificant with p-value < 0,001. It seems that age doesn’t explain whether or not a student is a high user of alcohol.
# compute odds ratios (OR)
OR <- coef(m) %>% exp
# compute confidence intervals (CI)
CI <- confint(m) %>% exp
## Waiting for profiling to be done...
# print out the odds ratios with their confidence intervals
cbind(OR, CI)
## OR 2.5 % 97.5 %
## (Intercept) 0.01237102 0.0003481843 0.3998333
## age 1.20373817 0.9852330457 1.4775972
## studytime 0.59545916 0.4321414645 0.8061535
## freetime 1.39118359 1.0938389338 1.7826010
## absences 1.08172440 1.0368164005 1.1336048
Odds ratio (OR) and the 95% confidence interval (CI) shows that those students who have a low study time are almost two times as likely to be a high user of alcohol than those studets who have higher study time. Students that have more freetime are also more likely to be a high user of alcohol. Also absences are positively correlated with high use of alcohol. Confidence interval shows that age is not statistically significant (because the interval contains 1) and other variables are.
Predictive power of the final logistic regression model is calculated without the statistically insignificant variable age.
#the model with glm() and without the age variable
m_final <- glm(high_use ~ studytime + freetime + absences, data = alc, family = "binomial")
summary(m_final)
##
## Call:
## glm(formula = high_use ~ studytime + freetime + absences, family = "binomial",
## data = alc)
##
## Deviance Residuals:
## Min 1Q Median 3Q Max
## -1.9420 -0.8332 -0.6450 1.1266 2.1537
##
## Coefficients:
## Estimate Std. Error z value Pr(>|z|)
## (Intercept) -1.34105 0.56337 -2.380 0.017293 *
## studytime -0.50496 0.15691 -3.218 0.001290 **
## freetime 0.32626 0.12379 2.636 0.008401 **
## absences 0.08324 0.02262 3.680 0.000233 ***
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## (Dispersion parameter for binomial family taken to be 1)
##
## Null deviance: 465.68 on 381 degrees of freedom
## Residual deviance: 426.47 on 378 degrees of freedom
## AIC: 434.47
##
## Number of Fisher Scoring iterations: 4
#predict and add the answer and the prediction to the data (alc)
probabilities <- predict(m_final, type = "response")
alc <- mutate(alc, probability = probabilities)
alc <- mutate(alc, prediction = probabilities > 0.5)
#tabulate the target variable versus the prediction
table("High use" = alc$high_use, "Prediction" = alc$prediction)
## Prediction
## High use FALSE TRUE
## FALSE 254 14
## TRUE 93 21
Table shows that the model predict 254 true negatives, 21 true positives, 14 false negatives and 93 false postives. This is sometimes called “confusion table”
table("High use" = alc$high_use, "Prediction" = alc$prediction) %>% prop.table() %>% addmargins()
## Prediction
## High use FALSE TRUE Sum
## FALSE 0.66492147 0.03664921 0.70157068
## TRUE 0.24345550 0.05497382 0.29842932
## Sum 0.90837696 0.09162304 1.00000000
Propabilities of the same table shows that 90,8% is predicted to be false but only 66,5% of them is correct. 9,2% is predicted to be true but 5,5% of them realy are students with high use of alcohol.
#a plot of 'high_use' versus 'probability' in 'alc'
g <- ggplot(alc, aes(x = probability, y = high_use, col = prediction))
g + geom_point()
#defining a loss function (mean prediction error)
loss_func <- function(class, prob) {
n_wrong <- abs(class - prob) > 0.5
mean(n_wrong)
}
#calling loss_func to compute the average number of wrong predictions in the (training) data
loss_func(class = alc$high_use, prob = alc$probability)
## [1] 0.2801047
The average number of wrong predictions in trainig data is 28%.
#computing the average number of wrong predictions in the (training) data
#loss_func(class = alc$high_use, prob = alc$probability)
#K-fold cross-validation
library(boot)
cv <- cv.glm(data = alc, cost = loss_func, glmfit = m_final, K = 10)
#average number of wrong predictions in the cross validation
cv$delta[1]
## [1] 0.2827225
10-fold cross-validation gives good estimate of the actual predictive power of the model. Low value = good.
In this exercise we use Boston data from MASS-library. This dataset contains information collected by the U.S Census Service concerning housing in the area of Boston Mass. Data includes 14 variables and 506 rows.
## [1] 506 14
## 'data.frame': 506 obs. of 14 variables:
## $ crim : num 0.00632 0.02731 0.02729 0.03237 0.06905 ...
## $ zn : num 18 0 0 0 0 0 12.5 12.5 12.5 12.5 ...
## $ indus : num 2.31 7.07 7.07 2.18 2.18 2.18 7.87 7.87 7.87 7.87 ...
## $ chas : int 0 0 0 0 0 0 0 0 0 0 ...
## $ nox : num 0.538 0.469 0.469 0.458 0.458 0.458 0.524 0.524 0.524 0.524 ...
## $ rm : num 6.58 6.42 7.18 7 7.15 ...
## $ age : num 65.2 78.9 61.1 45.8 54.2 58.7 66.6 96.1 100 85.9 ...
## $ dis : num 4.09 4.97 4.97 6.06 6.06 ...
## $ rad : int 1 2 2 3 3 3 5 5 5 5 ...
## $ tax : num 296 242 242 222 222 222 311 311 311 311 ...
## $ ptratio: num 15.3 17.8 17.8 18.7 18.7 18.7 15.2 15.2 15.2 15.2 ...
## $ black : num 397 397 393 395 397 ...
## $ lstat : num 4.98 9.14 4.03 2.94 5.33 ...
## $ medv : num 24 21.6 34.7 33.4 36.2 28.7 22.9 27.1 16.5 18.9 ...
| variable | description |
|---|---|
| crim | per capita crime rate by town |
| zn | proportion of residential land zoned for lots over 25,000 sq.ft. |
| indus | proportion of non-retail business acres per town |
| chas | Charles River dummy variable (= 1 if tract bounds river; 0 otherwise) |
| nox | nitrogen oxides concentration (parts per 10 million) |
| rm | average number of rooms per dwelling |
| age | proportion of owner-occupied units built prior to 1940 |
| dis | weighted mean of distances to five Boston employment centres |
| rad | index of accessibility to radial highways |
| tax | full-value property-tax rate per $10,000 |
| ptratio | pupil-teacher ratio by town |
| black | 1000(Bk - 0.63)^2 where Bk is the proportion of blacks by town |
| lstat | lower status of the population (percent) |
| medv | median value of owner-occupied homes in $1000 |
There are some very intresting distributions fo variables in the plot matrix. Variable rad has high and low values so the plot shows that the values are consenrated either side of the plot. VAriable *
Plotted correlation matrix shows that there is some high correlation between variables:
Correlation is quite clear between industrial areas (indus) and nitrogen oxides (nox). Industry adds pollution in the area. Industry seems to correlate also with variablrs like age, dis, ras and tax.
Nitrogen oxides (nox) correlations are very similar with industry (indus)
Crime rate (crim) seems to correlate with good accessibilitty to radial highways (rad) and value property (tax).
Old houses (age) and employment centers have also something common
summary(Boston)
## crim zn indus chas
## Min. : 0.00632 Min. : 0.00 Min. : 0.46 Min. :0.00000
## 1st Qu.: 0.08204 1st Qu.: 0.00 1st Qu.: 5.19 1st Qu.:0.00000
## Median : 0.25651 Median : 0.00 Median : 9.69 Median :0.00000
## Mean : 3.61352 Mean : 11.36 Mean :11.14 Mean :0.06917
## 3rd Qu.: 3.67708 3rd Qu.: 12.50 3rd Qu.:18.10 3rd Qu.:0.00000
## Max. :88.97620 Max. :100.00 Max. :27.74 Max. :1.00000
## nox rm age dis
## Min. :0.3850 Min. :3.561 Min. : 2.90 Min. : 1.130
## 1st Qu.:0.4490 1st Qu.:5.886 1st Qu.: 45.02 1st Qu.: 2.100
## Median :0.5380 Median :6.208 Median : 77.50 Median : 3.207
## Mean :0.5547 Mean :6.285 Mean : 68.57 Mean : 3.795
## 3rd Qu.:0.6240 3rd Qu.:6.623 3rd Qu.: 94.08 3rd Qu.: 5.188
## Max. :0.8710 Max. :8.780 Max. :100.00 Max. :12.127
## rad tax ptratio black
## Min. : 1.000 Min. :187.0 Min. :12.60 Min. : 0.32
## 1st Qu.: 4.000 1st Qu.:279.0 1st Qu.:17.40 1st Qu.:375.38
## Median : 5.000 Median :330.0 Median :19.05 Median :391.44
## Mean : 9.549 Mean :408.2 Mean :18.46 Mean :356.67
## 3rd Qu.:24.000 3rd Qu.:666.0 3rd Qu.:20.20 3rd Qu.:396.23
## Max. :24.000 Max. :711.0 Max. :22.00 Max. :396.90
## lstat medv
## Min. : 1.73 Min. : 5.00
## 1st Qu.: 6.95 1st Qu.:17.02
## Median :11.36 Median :21.20
## Mean :12.65 Mean :22.53
## 3rd Qu.:16.95 3rd Qu.:25.00
## Max. :37.97 Max. :50.00
All the variables are numerical so we can use scale()-function to scale whole data set.
## crim zn indus
## Min. :-0.419367 Min. :-0.48724 Min. :-1.5563
## 1st Qu.:-0.410563 1st Qu.:-0.48724 1st Qu.:-0.8668
## Median :-0.390280 Median :-0.48724 Median :-0.2109
## Mean : 0.000000 Mean : 0.00000 Mean : 0.0000
## 3rd Qu.: 0.007389 3rd Qu.: 0.04872 3rd Qu.: 1.0150
## Max. : 9.924110 Max. : 3.80047 Max. : 2.4202
## chas nox rm age
## Min. :-0.2723 Min. :-1.4644 Min. :-3.8764 Min. :-2.3331
## 1st Qu.:-0.2723 1st Qu.:-0.9121 1st Qu.:-0.5681 1st Qu.:-0.8366
## Median :-0.2723 Median :-0.1441 Median :-0.1084 Median : 0.3171
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.:-0.2723 3rd Qu.: 0.5981 3rd Qu.: 0.4823 3rd Qu.: 0.9059
## Max. : 3.6648 Max. : 2.7296 Max. : 3.5515 Max. : 1.1164
## dis rad tax ptratio
## Min. :-1.2658 Min. :-0.9819 Min. :-1.3127 Min. :-2.7047
## 1st Qu.:-0.8049 1st Qu.:-0.6373 1st Qu.:-0.7668 1st Qu.:-0.4876
## Median :-0.2790 Median :-0.5225 Median :-0.4642 Median : 0.2746
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.6617 3rd Qu.: 1.6596 3rd Qu.: 1.5294 3rd Qu.: 0.8058
## Max. : 3.9566 Max. : 1.6596 Max. : 1.7964 Max. : 1.6372
## black lstat medv
## Min. :-3.9033 Min. :-1.5296 Min. :-1.9063
## 1st Qu.: 0.2049 1st Qu.:-0.7986 1st Qu.:-0.5989
## Median : 0.3808 Median :-0.1811 Median :-0.1449
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.4332 3rd Qu.: 0.6024 3rd Qu.: 0.2683
## Max. : 0.4406 Max. : 3.5453 Max. : 2.9865
## [1] "matrix"
Scaling the data makes variables look as if they are in the same range. Variables like black and tax were before scaling hundred fold compared to some other variables.
Variable crim is the base of the new categorical variable crime.
| categories | quantile points |
|---|---|
| low | 0%-25% |
| med_low | 25%-50% |
| med_high | 50%-75% |
| high | 75%-100% |
Quantile points of the variable crim
## 0% 25% 50% 75% 100%
## -0.419366929 -0.410563278 -0.390280295 0.007389247 9.924109610
## crime
## low med_low med_high high
## 127 126 126 127
## zn indus chas nox
## Min. :-0.48724 Min. :-1.5563 Min. :-0.2723 Min. :-1.4644
## 1st Qu.:-0.48724 1st Qu.:-0.8668 1st Qu.:-0.2723 1st Qu.:-0.9121
## Median :-0.48724 Median :-0.2109 Median :-0.2723 Median :-0.1441
## Mean : 0.00000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.04872 3rd Qu.: 1.0150 3rd Qu.:-0.2723 3rd Qu.: 0.5981
## Max. : 3.80047 Max. : 2.4202 Max. : 3.6648 Max. : 2.7296
## rm age dis rad
## Min. :-3.8764 Min. :-2.3331 Min. :-1.2658 Min. :-0.9819
## 1st Qu.:-0.5681 1st Qu.:-0.8366 1st Qu.:-0.8049 1st Qu.:-0.6373
## Median :-0.1084 Median : 0.3171 Median :-0.2790 Median :-0.5225
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.4823 3rd Qu.: 0.9059 3rd Qu.: 0.6617 3rd Qu.: 1.6596
## Max. : 3.5515 Max. : 1.1164 Max. : 3.9566 Max. : 1.6596
## tax ptratio black lstat
## Min. :-1.3127 Min. :-2.7047 Min. :-3.9033 Min. :-1.5296
## 1st Qu.:-0.7668 1st Qu.:-0.4876 1st Qu.: 0.2049 1st Qu.:-0.7986
## Median :-0.4642 Median : 0.2746 Median : 0.3808 Median :-0.1811
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 1.5294 3rd Qu.: 0.8058 3rd Qu.: 0.4332 3rd Qu.: 0.6024
## Max. : 1.7964 Max. : 1.6372 Max. : 0.4406 Max. : 3.5453
## medv crime
## Min. :-1.9063 low :127
## 1st Qu.:-0.5989 med_low :126
## Median :-0.1449 med_high:126
## Mean : 0.0000 high :127
## 3rd Qu.: 0.2683
## Max. : 2.9865
Training set contains 80% of the data. 20% is in the test set.
## [1] 212 40 263 292 76 311 418 109 345 451 268 98 272 399 214 391 232
## [18] 458 341 218 204 110 8 71 482 374 417 113 186 339 420 65 406 365
## [35] 452 400 370 484 256 22 125 96 405 485 465 206 390 90 344 378 104
## [52] 267 3 412 442 137 181 332 304 154 230 282 407 179 389 361 260 402
## [69] 35 487 309 356 372 492 68 324 156 213 224 33 229 142 279 12 349
## [86] 84 306 53 52 444 130 82 112 416 2 114 45 495 287 252 502 203
## [103] 150 36 384 126 453 360 56 386 435 276 305 148 44 281 337 59 331
## [120] 19 239 102 328 295 78 355 99 95 48 430 443 163 419 144 242 498
## [137] 174 369 250 394 147 7 191 240 494 255 330 293 296 504 64 18 450
## [154] 32 427 188 461 346 437 183 323 415 380 192 46 426 264 197 106 159
## [171] 388 39 329 211 353 491 173 352 371 275 434 493 327 226 259 477 358
## [188] 488 28 247 223 245 500 121 258 476 238 464 446 131 180 319 132 235
## [205] 41 1 499 335 433 274 134 241 145 244 423 160 94 254 280 438 297
## [222] 269 178 381 404 457 63 129 480 17 127 208 167 16 342 27 200 408
## [239] 193 72 195 439 288 261 210 37 246 55 340 479 248 93 445 307 70
## [256] 486 501 61 336 201 66 138 202 456 471 118 14 300 334 289 122 81
## [273] 315 83 6 146 162 475 165 321 343 454 58 209 347 449 459 124 92
## [290] 11 290 21 74 30 440 441 425 317 77 243 169 168 448 215 194 205
## [307] 85 489 413 301 190 270 73 67 462 42 397 51 497 401 54 432 20
## [324] 348 143 233 299 170 350 43 227 25 175 101 396 219 228 9 221 385
## [341] 217 49 182 199 139 60 469 506 103 187 26 47 225 354 455 376 158
## [358] 97 207 470 403 123 320 4 249 421 266 472 184 468 322 220 325 359
## [375] 505 10 189 87 176 431 135 86 24 308 172 69 313 474 490 409 136
## [392] 177 133 161 151 251 75 222 285 79 88 398 185 265
First the linear discriminant analysis (LDA) is fitted to the train set. The new categorical variable crime is the target variable and all the other variables of the dataset are predictor variables.
After fitting we draw the LDA biplot with arrows.
## Call:
## lda(crime ~ ., data = train)
##
## Prior probabilities of groups:
## low med_low med_high high
## 0.2673267 0.2524752 0.2450495 0.2351485
##
## Group means:
## zn indus chas nox rm
## low 0.92988887 -0.8851749 -0.199419804 -0.8689772 0.3893444
## med_low -0.06939829 -0.2885883 -0.002135914 -0.5858947 -0.1058606
## med_high -0.37290116 0.1417439 0.165126514 0.3391934 0.1753501
## high -0.48724019 1.0172655 -0.106556426 1.0513782 -0.3075459
## age dis rad tax ptratio
## low -0.8938883 0.8844816 -0.6766767 -0.7240746 -0.41656009
## med_low -0.3264994 0.3921687 -0.5573888 -0.4963234 -0.07093611
## med_high 0.3840996 -0.3604385 -0.4250387 -0.3325394 -0.28926394
## high 0.8132644 -0.8444120 1.6366336 1.5129868 0.77903654
## black lstat medv
## low 0.3805531 -0.74981179 0.4770593
## med_low 0.3182189 -0.14605184 0.0155140
## med_high 0.1091458 -0.01501243 0.2334421
## high -0.8119674 0.94100560 -0.7557489
##
## Coefficients of linear discriminants:
## LD1 LD2 LD3
## zn 0.08275327 0.66552374 -0.91928543
## indus -0.01247700 -0.11388118 0.39160911
## chas -0.11640380 -0.12662684 0.09035990
## nox 0.43619153 -0.87462286 -1.45220640
## rm -0.12773636 -0.06753409 -0.07621988
## age 0.18129385 -0.27683817 0.03928607
## dis -0.09285831 -0.27135407 0.24957985
## rad 3.38566920 1.01740850 0.01350980
## tax 0.05939564 -0.12894880 0.42620550
## ptratio 0.12386274 -0.03072254 -0.29821914
## black -0.11701054 0.02675081 0.11721667
## lstat 0.24961933 -0.23912609 0.37370519
## medv 0.22650902 -0.48372771 -0.22968174
##
## Proportion of trace:
## LD1 LD2 LD3
## 0.9557 0.0333 0.0109
## [1] 3 1 3 1 2 3 4 2 1 4 3 2 2 4 2 4 3 4 1 1 1 3 2 2 4 4 4 2 1 1 4 1 4 3 4
## [36] 4 4 3 1 3 2 2 4 3 4 2 4 1 1 4 2 3 1 4 4 3 1 1 2 3 3 1 4 1 4 4 3 4 3 4
## [71] 3 2 4 2 1 3 3 2 3 3 3 3 1 2 1 1 1 1 1 4 3 1 2 4 1 2 2 3 1 2 1 1 3 1 4
## [106] 2 4 4 1 4 4 2 1 3 2 1 1 2 1 3 2 2 2 1 2 1 1 1 2 4 4 3 4 4 2 3 2 4 2 4
## [141] 3 2 2 2 2 1 1 1 2 1 2 3 4 3 4 1 4 1 4 2 3 4 4 1 2 4 3 1 2 3 4 2 1 2 1
## [176] 2 2 1 4 1 4 2 3 3 3 4 4 4 3 3 3 2 2 1 3 4 3 4 4 3 1 3 3 3 1 1 2 1 4 2
## [211] 3 2 3 2 4 3 1 3 2 4 1 3 1 4 4 4 2 3 4 3 3 2 3 3 1 3 1 4 2 2 1 4 1 3 3
## [246] 2 2 1 1 4 2 1 4 1 2 3 2 2 1 1 1 3 1 4 4 2 3 1 1 1 1 1 3 1 1 3 3 4 3 2
## [281] 1 4 1 2 1 4 4 2 1 2 1 3 2 3 4 4 4 3 2 2 3 3 4 3 1 1 1 2 4 1 2 2 2 1 4
## [316] 2 4 2 3 4 1 4 3 1 3 3 1 3 1 2 3 3 2 2 4 2 3 2 3 4 1 2 1 1 2 2 4 1 2 1
## [351] 3 2 3 1 4 4 3 2 2 4 4 2 3 1 2 4 3 4 2 4 2 2 3 4 2 2 2 1 1 4 3 1 3 1 3
## [386] 2 3 4 2 4 3 1 3 3 3 2 1 3 1 1 1 4 2 3
## predicted
## correct low med_low med_high high
## low 13 6 0 0
## med_low 3 17 4 0
## med_high 1 10 14 2
## high 0 0 0 32
Prediction were quite good. There was some errors in the middle of the range but classes low and especially high were good. Only one correct representative of high class was predicted to med_low class.
I’m going to calculate what is the optimal number of clusters for Boston data. First I reload and scale the data. Variables need to be scaled to get comparable distances between observation.
## crim zn indus
## Min. :-0.419367 Min. :-0.48724 Min. :-1.5563
## 1st Qu.:-0.410563 1st Qu.:-0.48724 1st Qu.:-0.8668
## Median :-0.390280 Median :-0.48724 Median :-0.2109
## Mean : 0.000000 Mean : 0.00000 Mean : 0.0000
## 3rd Qu.: 0.007389 3rd Qu.: 0.04872 3rd Qu.: 1.0150
## Max. : 9.924110 Max. : 3.80047 Max. : 2.4202
## chas nox rm age
## Min. :-0.2723 Min. :-1.4644 Min. :-3.8764 Min. :-2.3331
## 1st Qu.:-0.2723 1st Qu.:-0.9121 1st Qu.:-0.5681 1st Qu.:-0.8366
## Median :-0.2723 Median :-0.1441 Median :-0.1084 Median : 0.3171
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.:-0.2723 3rd Qu.: 0.5981 3rd Qu.: 0.4823 3rd Qu.: 0.9059
## Max. : 3.6648 Max. : 2.7296 Max. : 3.5515 Max. : 1.1164
## dis rad tax ptratio
## Min. :-1.2658 Min. :-0.9819 Min. :-1.3127 Min. :-2.7047
## 1st Qu.:-0.8049 1st Qu.:-0.6373 1st Qu.:-0.7668 1st Qu.:-0.4876
## Median :-0.2790 Median :-0.5225 Median :-0.4642 Median : 0.2746
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.6617 3rd Qu.: 1.6596 3rd Qu.: 1.5294 3rd Qu.: 0.8058
## Max. : 3.9566 Max. : 1.6596 Max. : 1.7964 Max. : 1.6372
## black lstat medv
## Min. :-3.9033 Min. :-1.5296 Min. :-1.9063
## 1st Qu.: 0.2049 1st Qu.:-0.7986 1st Qu.:-0.5989
## Median : 0.3808 Median :-0.1811 Median :-0.1449
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.4332 3rd Qu.: 0.6024 3rd Qu.: 0.2683
## Max. : 0.4406 Max. : 3.5453 Max. : 2.9865
Next I calculate the distances between observations and determinen the number of clusters.
One way to determine the number of clusters is to look how the total of within cluster sum of squares (WCSS) behaves when the number of clusters changes. WCSS was calculated from 1 to 15 clusters. The optimal number of clusters is when the total WCSS drops radically. It seems that in this case optimal number of clusters is two. However we are here to learn so I decided to analyse model with four clusters.
After determining the number of clusters I run the K-means alcorithm again.
It seems that when the data is divided to four clusters there is some clear differences in distriputions of several variables. Crim, zn, indus and blacks are variables were one can distinguish clear patterns between clusters. Some variables (rad & tax) show that sometimes 1 or 2 clusters make a clear distripution but observation of other two clusters are ambigious and there is no clear pattern to be regognised.
After loading the Boston dataset I scale it to get comparable distances.
## crim zn indus
## Min. :-0.419367 Min. :-0.48724 Min. :-1.5563
## 1st Qu.:-0.410563 1st Qu.:-0.48724 1st Qu.:-0.8668
## Median :-0.390280 Median :-0.48724 Median :-0.2109
## Mean : 0.000000 Mean : 0.00000 Mean : 0.0000
## 3rd Qu.: 0.007389 3rd Qu.: 0.04872 3rd Qu.: 1.0150
## Max. : 9.924110 Max. : 3.80047 Max. : 2.4202
## chas nox rm age
## Min. :-0.2723 Min. :-1.4644 Min. :-3.8764 Min. :-2.3331
## 1st Qu.:-0.2723 1st Qu.:-0.9121 1st Qu.:-0.5681 1st Qu.:-0.8366
## Median :-0.2723 Median :-0.1441 Median :-0.1084 Median : 0.3171
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.:-0.2723 3rd Qu.: 0.5981 3rd Qu.: 0.4823 3rd Qu.: 0.9059
## Max. : 3.6648 Max. : 2.7296 Max. : 3.5515 Max. : 1.1164
## dis rad tax ptratio
## Min. :-1.2658 Min. :-0.9819 Min. :-1.3127 Min. :-2.7047
## 1st Qu.:-0.8049 1st Qu.:-0.6373 1st Qu.:-0.7668 1st Qu.:-0.4876
## Median :-0.2790 Median :-0.5225 Median :-0.4642 Median : 0.2746
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean : 0.0000
## 3rd Qu.: 0.6617 3rd Qu.: 1.6596 3rd Qu.: 1.5294 3rd Qu.: 0.8058
## Max. : 3.9566 Max. : 1.6596 Max. : 1.7964 Max. : 1.6372
## black lstat medv clust
## Min. :-3.9033 Min. :-1.5296 Min. :-1.9063 Min. :1.000
## 1st Qu.: 0.2049 1st Qu.:-0.7986 1st Qu.:-0.5989 1st Qu.:2.000
## Median : 0.3808 Median :-0.1811 Median :-0.1449 Median :3.000
## Mean : 0.0000 Mean : 0.0000 Mean : 0.0000 Mean :2.674
## 3rd Qu.: 0.4332 3rd Qu.: 0.6024 3rd Qu.: 0.2683 3rd Qu.:3.000
## Max. : 0.4406 Max. : 3.5453 Max. : 2.9865 Max. :4.000
Original Boston dataset is now scaled and the result of K-means clustering is saved to the variable clust
Next the LDA is performed and the biplot with arrows is created
## Call:
## lda(clust ~ ., data = scaled_Boston)
##
## Prior probabilities of groups:
## 1 2 3 4
## 0.2114625 0.1304348 0.4308300 0.2272727
##
## Group means:
## crim zn indus chas nox rm
## 1 -0.3912182 1.2671159 -0.8754697 0.5739635 -0.7359091 0.9938426
## 2 1.4330759 -0.4872402 1.0689719 0.4435073 1.3439101 -0.7461469
## 3 -0.3894453 -0.2173896 -0.5212959 -0.2723291 -0.5203495 -0.1157814
## 4 0.2797949 -0.4872402 1.1892663 -0.2723291 0.8998296 -0.2770011
## age dis rad tax ptratio black
## 1 -0.6949417 0.7751031 -0.5965444 -0.6369476 -0.96586616 0.34190729
## 2 0.8575386 -0.9620552 1.2941816 1.2970210 0.42015742 -1.65562038
## 3 -0.3256000 0.3182404 -0.5741127 -0.6240070 0.02986213 0.34248644
## 4 0.7716696 -0.7723199 0.9006160 1.0311612 0.60093343 -0.01717546
## lstat medv
## 1 -0.8200275 1.11919598
## 2 1.1930953 -0.81904111
## 3 -0.2813666 -0.01314324
## 4 0.6116223 -0.54636549
##
## Coefficients of linear discriminants:
## LD1 LD2 LD3
## crim 0.18113078 -0.5012256 -0.60535205
## zn 0.43297497 -1.0486194 0.67406151
## indus 1.37753200 0.3016928 1.07034034
## chas -0.04307937 -0.7598229 -0.22448239
## nox 1.04674638 -0.3861005 -0.33268952
## rm -0.14912869 -0.1510367 0.67942589
## age -0.09897424 0.0523110 0.26285587
## dis 0.13139210 -0.1593367 -0.03487882
## rad 0.65824136 0.5189795 0.48145070
## tax 0.28903561 -0.5773959 0.10350513
## ptratio 0.22236843 0.1668597 -0.09181715
## black -0.42730704 0.5843973 0.89869354
## lstat 0.24320629 -0.6197780 -0.01119242
## medv 0.21961575 -0.9485829 -0.17065360
##
## Proportion of trace:
## LD1 LD2 LD3
## 0.7596 0.1768 0.0636
Biplot shows that variables indus, zn and medv are the most influencial separators for the clusters.
Colors of the both plots is based to four classes. It seems that K-means plot shows the different clusters more clearly than the plot that is based on the crime classification.